# -*- coding: utf-8 -*-
# @Time    : 2021/7/7 14:46
# @Author  : LuoTianHang

# ####################config.py 说明##########################
# this script is the config script for the whole project


import argparse

def parser_config():
    parser=argparse.ArgumentParser(description="人群密度估计")
    parser.add_argument("--project_name",type=str,default="crowd counting")
    parser.add_argument("--use_cuda",type=bool,default=True)
    parser.add_argument("--seed",type=int,default=123)

    parser.add_argument("--train_batch_size",type=int,default=1)
    parser.add_argument("--test_batch_size",type=int,default=1)

    parser.add_argument("--resume",type=bool,default=True)
    parser.add_argument("--pretrained_weight",type=str,default="./weights/best.pth")

    parser.add_argument("--lr",type=float,default=0.00001)
    parser.add_argument("--epochs",type=int,default=20000)

    return parser.parse_args()